#!/usr/bin/python
# -*- encoding: utf-8 -*-
'''
@File    :   
@Time    :   // ::
@Author  :   Malygos 
@Version :   1.0
@Contact :   751756061@qq.com
@WebSite :   https://gitee.com/malygos_l/python_homework
'''
# Start typing your code from here
'''将薪资格式化以便计算平均薪资'''

import re


# 格式化薪资，将薪资变成float形式
def format_salary(data):

    Date_conversion = 0  # 日期换算

    # 薪资以 元/年 元/月 元/天 元/小时 形式 最后统一为 元/月
    if data[-1] == '年':
        Date_conversion = 12
    elif data[-1] == '月':
        Date_conversion = 1
    elif data[-1] == '天':
        Date_conversion = 1 / 30
    elif data[-1] == '时':
        Date_conversion = 1 / 720
    # print(Date_conversion)

    salary_str = '[0-9]*\.*[0-9]*'    # 匹配薪资中数字部分的字符串
    Salary_conversion = 0   # 薪资换算

    # 按 万（千）元上/年（月、天、小时） 千（万）/年（月、天、小时） 分类格式化薪资
    if data[-3] == '千':
        Salary_conversion = 1000
    elif data[-3] == '万':
        Salary_conversion = 10000
    elif data[-3] == '上':
        if data[-5] == '万':
            Salary_conversion = 10000
        elif data[-5] == '千':
            Salary_conversion = 1000
        return float(re.match(
            salary_str, data).group(0)) * Salary_conversion / Date_conversion
    elif data[-3] == '下':
        if data[-5] == '万':
            Salary_conversion = 10000
        elif data[-5] == '千':
            Salary_conversion = 1000
        return float(re.match(
            salary_str, data).group(0)) * Salary_conversion / Date_conversion

    # 将工资范围折中取半
    salary = re.findall(salary_str, data)
    # print(salary)

    # 有工资范围
    if salary[2] != '':
        avg_salary = (float(salary[0]) + float(salary[2])) / 2
    # 有确定值
    else:
        return float(salary[0]) * Salary_conversion / Date_conversion
    
    # 以float的形式返回格式化后的薪资
    return avg_salary * Salary_conversion / Date_conversion
